package com.skytech.component.crawler.logger;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Service;

import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

@Service
public class LoggerService {


    public JSONObject getLogger(ResponseEntity<JSONObject> responseEntity) {

        JSONArray jsonArray = responseEntity.getBody().getJSONObject("hits").getJSONArray("hits");
        String total= responseEntity.getBody().getJSONObject("hits").get("total").toString();
        List<JSONObject> jsonObjectList=new ArrayList<>();
        for (Object js : jsonArray) {
            JSONObject object = JSONObject.parseObject(js.toString());
            JSONObject jsonObject1 = object.getJSONObject("_source");

            jsonObject1.put("time",UTCStringtODefaultString(object.getJSONObject("_source").get("@timestamp").toString()));
            jsonObjectList.add(jsonObject1);
        }

        JSONObject jsonObjectNew=new JSONObject();
        jsonObjectNew.put("data",jsonObjectList);
        jsonObjectNew.put("total",total);


        return jsonObjectNew;
    }

    public String UTCStringtODefaultString(String UTCString) {
        try{
            UTCString = UTCString.replace("Z", " UTC");
            SimpleDateFormat utcFormat = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSS Z");
            SimpleDateFormat defaultFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            Date date = utcFormat.parse(UTCString);
            return defaultFormat.format(date);
        } catch(ParseException pe)
        {
            pe.printStackTrace();
            return null;
        }
    }


    public List<JSONObject> getLoggerBy(ResponseEntity<JSONObject> responseEntity){
        JSONArray jsonArray=responseEntity.getBody().getJSONObject("aggregations").getJSONObject("articles_over_time").getJSONArray("buckets");
        List<JSONObject> jsonObjectList=new ArrayList<>();
        for (Object js : jsonArray) {
            JSONObject object = JSONObject.parseObject(js.toString());
            JSONObject jsonObject1 = new JSONObject();

            jsonObject1.put("time",UTCStringtODefaultString(object.getString("key_as_string")));
           jsonObject1.put("count",object.get("doc_count").toString());
            jsonObjectList.add(jsonObject1);
        }

        return jsonObjectList;
    }
}
